package com.leetcode;

import java.util.HashMap;
import java.util.Map;

/**
 * 两数之和
 *
 * 给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。
 * 假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
 *
 * 时间复杂度小于 O(n2) 的算法
 * @author winson
 * @create 2022-03-25 14:53
 */
public class SumOfTwoNumberSolution {

    public int[] twoSum(int[] nums, int target) {
        /**
         * 暴力解法，通过遍历判断另外一个数是否存在，另外一个数 = target - num[i]
          */
        int length = nums.length;
        for (int i = 0; i < length; i++) {
            int anotherNum = target - nums[i];
            /**
             * 遍历剩余数组，查找另外一个数
             */
            for (int j = i + 1; j < length; j++) {
                if (anotherNum == nums[j]) {
                    return new int[]{i, j};
                }
            }
        }
        return null;
    }

    /**
     * 优化思考，如何快速查找另外一个数，通过哈希快速查找
     */
    public int[] twoSumOptimized(int[] nums, int target) {
        Map<Integer, Integer> map = new HashMap<>(nums.length);
        for (int i = 0; i < nums.length; i++) {
            /**
             * 判断另外一个数是否存在
             */
            if (map.containsKey(target - nums[i])) {
                return new int[]{i, map.get(target - nums[i])};
            } else {
                map.put(nums[i], i);
            }
        }
        return null;
    }

    public static void main(String[] args) {
        /**
         * 输入：nums = [2,7,11,15], target = 9
         * 输出：[0,1]
         */
        SumOfTwoNumberSolution solution = new SumOfTwoNumberSolution();
        int[] nums = new int[] {2,7,11,15};
        int[] res1 = solution.twoSum(nums, 9);
        for (int n : res1) {
            System.out.println(n);
        }
        int[] res2 = solution.twoSumOptimized(nums, 9);
        for (int n : res2) {
            System.out.println(n);
        }
    }
}
